import os
import django

# 设置Django设置模块
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'demo.settings')
django.setup()

# 导入User模型
from django.contrib.auth.models import User

# 检查admin用户是否存在
try:
    user = User.objects.get(username='admin')
    print('admin用户已存在！')
    # 可选：更新现有admin用户的密码
    user.set_password('admin')
    user.save()
    print('admin用户的密码已更新为: admin')
except User.DoesNotExist:
    # 创建admin超级用户
    user = User.objects.create_superuser('admin', 'admin@example.com', 'admin')
    user.save()
    print('admin超级用户已创建成功！')
    print('用户名: admin')
    print('密码: admin')
    print('邮箱: admin@example.com')